import torch
from ultralytics.nn.tasks import DetectionModel
from safetensors.torch import load_file

def convert_safetensor_to_pt():
    tensor_dict = load_file("weights/icon_detect/model.safetensors")
    
    model = DetectionModel('weights/icon_detect/model.yaml')
    model.load_state_dict(tensor_dict)
    torch.save({'model': model}, 'weights/icon_detect/best.pt')


